from keras.models import Sequential
from keras.layers import Dense

model = Sequential()
model.add(Dense(500,input_dim=image_height*image_width,activation='relu'))
model.add(Dense(10,  activation='softmax'))
print(model.summary())
EPOCHS = 50
model.compile(loss='categorical_crossentropy', optimizer='adam',
              metrics=['accuracy'])
history = model.fit(X_train, y_train, validation_data=(X_test, y_test), epochs=EPOCHS, batch_size=256, verbose=0)
